import os
import random

from dotenv import load_dotenv
# 通过 pip install 'volcengine-python-sdk[ark]' 安装方舟SDK
from volcenginesdkarkruntime import Ark
load_dotenv()

# 请确保您已将 API Key 存储在环境变量 ARK_API_KEY 中
# 初始化Ark客户端，从环境变量中读取您的API Key
client = Ark(
    # 此为默认路径，您可根据业务所在地域进行配置
    base_url="https://ark.cn-beijing.volces.com/api/v3",
    # 从环境变量中获取您的 API Key。此为默认方式，您可根据需要进行修改
    api_key=os.environ.get("ARK_API_KEY"),
)

imagesResponse = client.images.generate(
    model="doubao-seededit-3-0-i2i-250628",
    prompt="改成爱心形状的泡泡",
    image="https://ark-project.tos-cn-beijing.volces.com/doc_image/seededit_i2i.jpeg",
    seed=random.randint(1, 10000),  # 使用相同的seed值可以得到一致的结果，便于复现
    guidance_scale=8.5,   # 指导强度参，数控制模型遵循prompt提示的程度，值越高，生成结果越贴近prompt描述，通常取值范围在1-20之间
    size="adaptive",  # 输出图像尺寸设置："adaptive"表示自适应尺寸
    watermark=False    # 是否添加水印
)

print(imagesResponse.data[0].url)